# coding=utf-8

import os

model_name_list = [
    'deepseek_r1_vllm',
    'vemlp-deepseek-r1-enc',
    '/data/model/deepseek_r1',
    '/data1/model/deepseek-r1-qint4'
]
model_name = '/data1/model/deepseek-r1-qint4'


url_list = [
    'http://127.0.0.1:40001/v1/chat/completions',
    'http://9.149.135.102:40001/v1/chat/completions',
]
url = 'http://9.149.135.102:40001/v1/chat/completions'


data_file_list = []

current_directory = os.getcwd()
# 列出当前目录下的所有文件
files_in_directory = os.listdir(current_directory)
# 过滤出所有.ndjson文件
data_file_list = [file for file in files_in_directory if file.endswith('.ndjson')]
data_file = 'yb-r1-20250220-1600-1700-5000.ndjson'

share_data_list = [0, 1, 2]
share_data = 1

sample_type_list = [0, 1]
sample_type = 0

api_key = os.getenv('API_KEY')

use_uuid_list = [0, 1]
use_uuid = 1

